package com.itheima.controller;

import com.alibaba.dubbo.config.annotation.Reference;
import com.itheima.constant.MessageConstant;
import com.itheima.entity.Result;
import com.itheima.service.MemberService;
import com.itheima.service.OrderService;
import com.itheima.service.ReportService;
import org.apache.poi.xssf.usermodel.XSSFRow;
import org.apache.poi.xssf.usermodel.XSSFSheet;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
import org.springframework.util.CollectionUtils;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.math.BigDecimal;
import java.time.LocalDate;
import java.time.format.DateTimeFormatter;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * @Author: bowen
 * @Date: 2019/7/2 17:10
 * @Description:获取报表数据
 */
@RestController
@RequestMapping("/report")
public class ReportController {
	@Reference
	MemberService memberService;

	@Reference
	OrderService orderService;

	@Reference
	ReportService reportService;

	/**
	 *  @描述：获取会员数据
	 */
	@RequestMapping("/getMemberReport")
	public Result getMemberReport() {
		Map map = new HashMap();
		// 计算月份
		LocalDate localDate = LocalDate.now().minusMonths(12);
		List<String> months = new ArrayList<>();
		for (int i = 1; i <=12; i++) {

			LocalDate temp = localDate.plusMonths(i);
			months.add(temp.toString().substring(0,7));
		}
		map.put("months",months);
		List<Integer> memberCounts = memberService.findCountByMonth(months);
		map.put("memberCount",memberCounts);
		return new Result(true, MessageConstant.GET_MEMBER_NUMBER_REPORT_SUCCESS,map);
	}

	/**
	 *  @描述：获取套餐数据
	 */
	@RequestMapping("/getSetmealReport")
	public Result getSetmealReport() {
		Map<String,Object> map = new HashMap();
		List<String> setmealNames = new ArrayList();
		List<Map> setmealCount = new ArrayList();

		setmealCount = orderService.getSetmealReport();
		setmealCount.forEach(item->{
			setmealNames.add(String.valueOf(item.get("name")));
		});


		map.put("setmealNames",setmealNames);
		map.put("setmealCount",setmealCount);
	  return new Result(true,MessageConstant.GET_SETMEAL_COUNT_REPORT_SUCCESS,map);
	}
    /**
     *  @描述：获取运营数据
     */
    @RequestMapping("/getBusinessReportData")
	public Result getBusinessReportData() {

	    Map<String,Object>  map= reportService.getBusinessReportData();
	  return new Result(true,MessageConstant.GET_BUSINESS_REPORT_SUCCESS,map);
	}

	/**
	 *  @描述：导出Excel
	 */
	@RequestMapping("/exportBusinessReport")
	public Result exportBusinessReport(HttpServletRequest request, HttpServletResponse response) {
        // 获取运营数据
		Map<String,Object>  map= reportService.getBusinessReportData();
		//取出返回结果数据，准备将报表数据写入到Excel文件中
		String reportDate = DateTimeFormatter.ofPattern("yyyy-MM-dd").format((LocalDate)map.get("reportDate"));
		Integer todayNewMember = (Integer) map.get("todayNewMember");
		Integer totalMember = (Integer) map.get("totalMember");
		Integer thisWeekNewMember = (Integer) map.get("thisWeekNewMember");
		Integer thisMonthNewMember = (Integer) map.get("thisMonthNewMember");
		Integer todayOrderNumber = (Integer) map.get("todayOrderNumber");
		Integer thisWeekOrderNumber = (Integer) map.get("thisWeekOrderNumber");
		Integer thisMonthOrderNumber = (Integer) map.get("thisMonthOrderNumber");
		Integer todayVisitsNumber = (Integer) map.get("todayVisitsNumber");
		Integer thisWeekVisitsNumber = (Integer) map.get("thisWeekVisitsNumber");
		Integer thisMonthVisitsNumber = (Integer) map.get("thisMonthVisitsNumber");
		List<Map> hotSetmeal = (List<Map>) map.get("hotSetmeal");

		//加载模板对象

		//获得Excel模板文件绝对路径
		String temlateRealPath = request.getSession().getServletContext().getRealPath("template") +
				File.separator + "report_template.xlsx";
		try {
			XSSFWorkbook xssfWorkbook = new XSSFWorkbook(new FileInputStream(new File(temlateRealPath)));

			//xssfWorkbook.getSheetAt(0).getRow(2).getCell(5).setCellValue("2019-01-01");
            XSSFRow row;
			XSSFSheet sheet = xssfWorkbook.getSheetAt(0);

			row = sheet.getRow(2);
			row.getCell(5).setCellValue(reportDate);//日期

			row = sheet.getRow(4);
			row.getCell(5).setCellValue(todayNewMember);//新增会员数（本日）
			row.getCell(7).setCellValue(totalMember);//总会员数

			row = sheet.getRow(5);
			row.getCell(5).setCellValue(thisWeekNewMember);//本周新增会员数
			row.getCell(7).setCellValue(thisMonthNewMember);//本月新增会员数

			row = sheet.getRow(7);
			row.getCell(5).setCellValue(todayOrderNumber);//今日预约数
			row.getCell(7).setCellValue(todayVisitsNumber);//今日到诊数

			row = sheet.getRow(8);
			row.getCell(5).setCellValue(thisWeekOrderNumber);//本周预约数
			row.getCell(7).setCellValue(thisWeekVisitsNumber);//本周到诊数

			row = sheet.getRow(9);
			row.getCell(5).setCellValue(thisMonthOrderNumber);//本月预约数
			row.getCell(7).setCellValue(thisMonthVisitsNumber);//本月到诊数


			if(!CollectionUtils.isEmpty(hotSetmeal)){
				int num = 12;
				for(Map tempMap : hotSetmeal){//热门套餐
					String name = (String) tempMap.get("name");
					Long setmeal_count = (Long) tempMap.get("setmeal_count");
					BigDecimal proportion = (BigDecimal) tempMap.get("proportion");
					row = sheet.getRow(num ++);
					row.getCell(4).setCellValue(name);//套餐名称
					row.getCell(5).setCellValue(setmeal_count);//预约数量
					row.getCell(6).setCellValue(proportion.doubleValue());//占比
				}
				//hotSetmeal.forEach(item->{
				//
				//	XSSFRow tempRow;
				//	tempRow = sheet.getRow(num);
				//	tempRow.getCell(4).setCellValue(String.valueOf(item.get("name")));
				//	tempRow.getCell(5).setCellValue(String.valueOf(item.get("setmeal_count")));
				//	tempRow.getCell(6).setCellValue(String.valueOf(item.get("proportion")));
				//	num ++;
				//
				//});





			}







			ServletOutputStream out = response.getOutputStream();
			response.setContentType("application/vnd.ms-excel");
			response.setHeader("content-Disposition", "attachment;filename=report.xlsx");



			xssfWorkbook.write(out);

			out.flush();
			out.close();
			xssfWorkbook.close();




		} catch (Exception e) {
			e.printStackTrace();
			return new Result(false, MessageConstant.GET_BUSINESS_REPORT_FAIL);
		}




		return null;
	}


	public static void main(String[] args) {
		LocalDate localDate = LocalDate.now().minusMonths(12);
		List<String> list = new ArrayList<>();
		for (int i = 1; i <=12; i++) {

			LocalDate temp = localDate.plusMonths(i);
			list.add(temp.toString().substring(0,7));
		}

		list.forEach(item->{
			System.out.println(item);
		});

	}
}
